package org.yi.utils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.Namespace;
import org.dom4j.QName;

public class XmlUtils {

	/**
	 * parse xml to hashmap
	 * @param xml
	 * @return
	 * @throws DocumentException
	 */
	public static Map<String, Object> toMap(String xml)
			throws DocumentException {
		Document doc = DocumentHelper.parseText(xml);
		Element root = doc.getRootElement();
		Map<String, Object> map = new HashMap<String, Object>();
		ele2Map(map, root); 
		return map;
	}
	
	/**
	 * parse sub node
	 * @param map
	 * @param root
	 */
	@SuppressWarnings("unchecked")
	private static void ele2Map(Map<String, Object> map, Element root) {
		// 获得当前节点的子节点
		List<Element> elements = root.elements();
		if (elements.size() == 0) {
			// 没有子节点说明当前节点是叶子节点，直接取值即可
			map.put(root.getName(), root.getText());
		} else if (elements.size() == 1) {
			// 只有一个子节点说明不用考虑list的情况，直接继续递归即可
			Map<String, Object> tempMap = new HashMap<String, Object>();
			ele2Map(tempMap, elements.get(0));
			map.put(root.getName(), tempMap);
		} else {
			// 多个子节点的话就得考虑list的情况了，比如多个子节点有节点名称相同的
			// 构造一个map用来去重
			Map<String, Object> tempMap = new HashMap<String, Object>();
			for (Element element : elements) {
				tempMap.put(element.getName(), null);
			}
			Set<String> keySet = tempMap.keySet();
			for (String string : keySet) {
				Namespace namespace = elements.get(0).getNamespace();
				List<Element> elements2 = root.elements(new QName(string, namespace));
				// 如果同名的数目大于1则表示要构建list
				if (elements2.size() > 1) {
					List<Map<String,Object>> list = new ArrayList<Map<String,Object>>();
					for (Element element : elements2) {
						Map<String, Object> tempMap1 = new HashMap<String, Object>();
						ele2Map(tempMap1, element);
						list.add(tempMap1);
					}
					map.put(string, list);
				} else {
					// 同名的数量不大于1则直接递归去
					Map<String, Object> tempMap1 = new HashMap<String, Object>();
					ele2Map(tempMap1, elements2.get(0));
					map.put(string, tempMap1);
				}
			}
		}
	}

}
